import os
# import torch
# import torchvision.transforms as transforms
from PIL import Image

# 定义原始图片文件夹和目标图片文件夹路径
input_folder = 'E:\Desktop\机械臂项目\chess'
output_folder = 'chess2'

# 确保输出文件夹存在，如果不存在则创建
if not os.path.exists(output_folder):
    os.makedirs(output_folder)

# 定义目标图像大小
target_width = 600
target_height = 600

# 遍历原始图片文件夹中的所有图片
for filename in os.listdir(input_folder):
    if filename.endswith('.jpg') or filename.endswith('.png'):
        # 读取原始图像
        image_path = os.path.join(input_folder, filename)
        image = Image.open(image_path)

        # 计算需要填充的位置和大小
        pad_left = (target_width - image.width) // 2
        pad_top = (target_height - image.height) // 2

        # 创建新的图像，并使用白色填充
        new_image = Image.new("RGB", (target_width, target_height), color="white")

        # 将原始图像放入新图像中
        new_image.paste(image, (pad_left, pad_top))

        # 保存调整大小后的图像到目标文件夹
        output_path = os.path.join(output_folder, filename)
        new_image.save(output_path)

print("图像处理完成！")
